package com.czk.netty.netty.simple;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.CharsetUtil;
import io.netty.util.concurrent.DefaultEventExecutorGroup;
import io.netty.util.concurrent.EventExecutorGroup;

import java.nio.charset.StandardCharsets;
import java.util.concurrent.TimeUnit;

/**
 * 自定义handler需要继承netty规定好的某个handler适配器
 * 自定义的handler才能成为handler
 *
 * @Author:ChenZhangKun
 * @Date: 2021/1/13 22:43
 */
public class NettyServerHandler extends ChannelInboundHandlerAdapter {
    /**
     * 创建线程池
     * 将耗时任务提交到业务线程，就不会阻塞netty提供的io线程
     */
    static final EventExecutorGroup group = new DefaultEventExecutorGroup(16);

    /**
     * ChannelHandlerContext 上下文对象，含有管道pipeline,通道channel,地址等
     * Object 客户端发送的数据，默认是Object的形式
     *
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        // 当前线程名称
        System.out.println(Thread.currentThread().getName());
        // 创建异步线程执行任务
        group.submit(() -> {
            // 接收客户端消息
            ByteBuf buf = (ByteBuf) msg;
            byte[] bytes = new byte[buf.readableBytes()];
            buf.readBytes(bytes);
            try {
                String str = new String(bytes, StandardCharsets.UTF_8);
                // 休眠
                Thread.sleep(5 * 1000);
                System.out.println(str);
                System.out.println("线程的名称是" + Thread.currentThread().getName());
                // 回复消息 此时会判断是否是io线程池的线程，会调用io线程池的线程线程将消息写回
                ctx.writeAndFlush(Unpooled.copiedBuffer("hello 客户端", StandardCharsets.UTF_8));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });


        // 有非常耗时的任务时，需要异步执行，加入该channel对应的NioEventLoop taskQueue中
        // 方案一
        ctx.channel().eventLoop().execute(() -> {
            try {
                TimeUnit.SECONDS.sleep(10);
                ctx.writeAndFlush(Unpooled.copiedBuffer("hello czk", CharsetUtil.UTF_8));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });
        // 30s后才会发送回去
        ctx.channel().eventLoop().execute(() -> {
            try {
                TimeUnit.SECONDS.sleep(20);
                ctx.writeAndFlush(Unpooled.copiedBuffer("hello nzq", CharsetUtil.UTF_8));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });

        // 方式二：用户自定义定时任务 该任务提交到scheduleTaskQueue
        ctx.channel().eventLoop().schedule(() -> {
            try {
                TimeUnit.SECONDS.sleep(5);
                ctx.writeAndFlush(Unpooled.copiedBuffer("Durant", CharsetUtil.UTF_8));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }, 5, TimeUnit.SECONDS);

      /*  // 依次分配线程池里面的线程来进行处理
        System.out.println("服务器读取线程"+Thread.currentThread().getName());
        System.out.println(ctx);
        // 将msg转成ByteBuf
        ByteBuf buf = (ByteBuf) msg;
        System.out.println("客户端发送的2消息是 " + buf.toString(CharsetUtil.UTF_8));
        // 客户端地址
        System.out.println(ctx.channel().remoteAddress());*/

    }

    /**
     * 读取完毕写回操作
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
        // 将数据写入缓存并刷新
        // 一般讲对发送的数据进行编码
        ctx.writeAndFlush(Unpooled.copiedBuffer("hello,客户端", CharsetUtil.UTF_8));
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        System.out.println(cause.getMessage());
        // 关闭连接
        ctx.close();
    }
}
